package com.bolt.common.utils;

import com.bolt.convention.exception.UtilException;
import net.sourceforge.pinyin4j.PinyinHelper;
import net.sourceforge.pinyin4j.format.HanyuPinyinCaseType;
import net.sourceforge.pinyin4j.format.HanyuPinyinOutputFormat;
import net.sourceforge.pinyin4j.format.HanyuPinyinToneType;
import net.sourceforge.pinyin4j.format.HanyuPinyinVCharType;
import net.sourceforge.pinyin4j.format.exception.BadHanyuPinyinOutputFormatCombination;

public class PinyinUtil {

    private static final String CHINESE_REGEX = "[\\u4e00-\\u9fa5]";

    /**
     * 获得全局单例的拼音引擎
     *
     * @return 全局单例的拼音引擎
     */
    private static HanyuPinyinOutputFormat getFormat() {
        //设置汉子拼音输出的格式
        HanyuPinyinOutputFormat format = new HanyuPinyinOutputFormat();
        // 小写
        format.setCaseType(HanyuPinyinCaseType.LOWERCASE);
        // 不加声调
        format.setToneType(HanyuPinyinToneType.WITHOUT_TONE);
        // 'ü' 使用 "v" 代替
        format.setVCharType(HanyuPinyinVCharType.WITH_V);
        return format;
    }

    private static String[] getPinyinStringArray(String str) {
        final int strLen = str.length();
        final String[] result = new String[strLen];
        try {
            for (int i = 0; i < strLen; i++) {
                final String[] pinyinStringArray = PinyinHelper.toHanyuPinyinStringArray(str.charAt(i), getFormat());
                if (ArrayUtil.isEmpty(pinyinStringArray)) {
                    result[i] = Character.toString(str.charAt(i));
                } else {
                    result[i] = pinyinStringArray[0];
                }
            }
        } catch (BadHanyuPinyinOutputFormatCombination e) {
            throw new UtilException(e);
        }
        return result;

    }

    /**
     * 如果c为汉字，则返回大写拼音；如果c不是汉字，则返回String.valueOf(c)
     *
     * @param c 任意字符，汉字返回拼音，非汉字原样返回
     * @return 汉字返回拼音，非汉字原样返回
     */
    public static String getPinyin(char c) {
        String result;
        try {
            String[] results = PinyinHelper.toHanyuPinyinStringArray(c, getFormat());
            result = ArrayUtil.isEmpty(results) ? String.valueOf(c) : results[0];
        } catch (BadHanyuPinyinOutputFormatCombination e) {
            result = String.valueOf(c);
        }
        return result;
    }

    /**
     * 将输入字符串转为拼音
     *
     * @param str 任意字符，汉字返回拼音，非汉字原样返回
     * @return 汉字返回拼音，非汉字原样返回
     */
    public static String getPinyin(String str, boolean upperFirst) {
        final String[] pinyinStringArray = getPinyinStringArray(str);
        final StringBuilder result = new StringBuilder();
        for (String s : pinyinStringArray) {
            result.append(StrUtil.upperFirst(s));
        }
        return result.toString();
    }

    /**
     * 将输入字符串转为拼音，以字符为单位插入分隔符
     *
     * @param str       任意字符，汉字返回拼音，非汉字原样返回
     * @param separator 每个字拼音之间的分隔符
     * @return 汉字返回拼音，非汉字原样返回
     */
    public static String getPinyin(String str, String separator) {
        return ArrayUtil.join(getPinyinStringArray(str), separator);
    }


    /**
     * 将输入字符串转为拼音首字母，其它字符原样返回
     *
     * @param c 任意字符，汉字返回拼音，非汉字原样返回
     * @return 汉字返回拼音，非汉字原样返回
     */
    public static char getFirstLetter(char c) {
        return getPinyin(c).charAt(0);
    }

    /**
     * 将输入字符串转为拼音首字母，其它字符原样返回
     *
     * @param str 任意字符，汉字返回拼音，非汉字原样返回
     * @return 汉字返回拼音，非汉字原样返回
     */
    public static String getFirstLetter(String str) {
        final StringBuilder result = new StringBuilder();
        final String[] pinyinStringArray = getPinyinStringArray(str);
        for (String s : pinyinStringArray) {
            final String first = String.valueOf(s.length() > 0 ? s.charAt(0) : StrUtil.EMPTY);
            result.append(first);
        }
        return result.toString();
    }

    /**
     * 将输入字符串转为拼音首字母，其它字符原样返回
     *
     * @param str       任意字符，汉字返回拼音，非汉字原样返回
     * @param separator 分隔符
     * @return 汉字返回拼音，非汉字原样返回
     */
    public static String getFirstLetter(String str, String separator) {
        final StringBuilder result = new StringBuilder();
        final String splitSeparator = StrUtil.isEmpty(separator) ? "#" : separator;
        final String[] pinyinStringArray = getPinyinStringArray(str);
        boolean isFirst = true;
        final int strLen = pinyinStringArray.length;
        for (int i = 0; i < strLen; i++) {
            if (isFirst) {
                isFirst = false;
            } else {
                result.append(separator);
            }
            String s = pinyinStringArray[i];
            if (s.length() > 0) {
                result.append(s.charAt(0));
            } else {
                result.append(StrUtil.EMPTY);
            }
        }
        return result.toString();
    }

    /**
     * 是否为中文字符
     *
     * @param c 字符
     * @return 是否为中文字符
     */
    public static boolean isChinese(char c) {
        return '〇' == c || String.valueOf(c).matches(CHINESE_REGEX);
    }

    public static void main(String[] args) {
        //汉字转拼单首字大写
        System.out.println(PinyinUtil.getPinyin("北京慧典云医", true));
        //将输入字符串转为拼音首字母
        System.out.println(PinyinUtil.getFirstLetter("北京慧典云医"));
        //将输入字符串转为拼音首字母 加拼接
        System.out.println(PinyinUtil.getFirstLetter("北京慧典云医", "-"));
    }
}